/**
 * This program is free software, you can redistribute it and/or modify.
 * Copyright (c) 2025 Huawei Technologies Co., Ltd.
 * This file is a part of the CANN Open Software.
 * Licensed under CANN Open Software License Agreement Version 2.0 (the "License").
 * Please refer to the License for details. You may not use this file except in compliance with the License.
 * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
 * See LICENSE in the root of the software repository for the full text of the License.
 */

#ifndef _DIAG_V2_TILING_H_
#define _DIAG_V2_TILING_H_

#include "kernel_tiling/kernel_tiling.h"

struct ChamferDistanceGradTilingDataTest {
    uint32_t batch_size = 0;     // 1
    uint32_t num = 0;            // 2
    uint64_t ub_size = 0;        // 3
    uint32_t task_per_core = 0;  // 4
    uint32_t core_used = 0;      // 5
    uint32_t task_tail_core = 0; // 6
};

#define DTYPE_X int64_t

#pragma pack(1)

#pragma pack()

#define CONVERT_TILING_DATA(tilingStruct, tilingDataPointer, tilingPointer) \
    __ubuf__ tilingStruct* tilingDataPointer =                              \
        reinterpret_cast<__ubuf__ tilingStruct*>((__ubuf__ uint8_t*)(tilingPointer));

#define INIT_TILING_DATA(tilingStruct, tilingDataPointer, tilingPointer) \
    CONVERT_TILING_DATA(tilingStruct, tilingDataPointer, tilingPointer);

#define GET_TILING_DATA(tilingData, tilingPointer)                                     \
    ChamferDistanceGradTilingDataTest tilingData;                                          \
    INIT_TILING_DATA(ChamferDistanceGradTilingDataTest, tilingDataPointer, tilingPointer); \
    (tilingData).batch_size = tilingDataPointer->batch_size;                           \
    (tilingData).num = tilingDataPointer->num;                                         \
    (tilingData).ub_size = tilingDataPointer->ub_size;                                 \
    (tilingData).task_per_core = tilingDataPointer->task_per_core;                     \
    (tilingData).core_used = tilingDataPointer->core_used;                             \
    (tilingData).task_tail_core = tilingDataPointer->task_tail_core;
#endif